
package com.yeye.util;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.TypeReference;
import org.springframework.http.HttpStatus;

import java.util.HashMap;
import java.util.Map;

/**
 * 返回数据
 *
 * @author Mark sunlightcs@gmail.com
 */
public class R extends HashMap<String, Object> {
  private static final long serialVersionUID = 1L;

  public R() {
    put("code", 0);
    put("msg", "success");
  }

  public <T> T getData(String name, TypeReference<T> tTypeReference) {
    Object data = get(name);
    String s = JSON.toJSONString(data);
    T t = JSON.parseObject(s, tTypeReference);
    return t;
  }

  public static R error() {
    return error(HttpStatus.INTERNAL_SERVER_ERROR.value(), "未知异常，请联系管理员");
  }

  public static R error(String msg) {
    return error(HttpStatus.INTERNAL_SERVER_ERROR.value(), msg);
  }

  public static R error(int code, String msg) {
    R r = new R();
    r.put("code", code);
    r.put("msg", msg);
    return r;
  }

  public static R ok(String msg) {
    R r = new R();
    r.put("msg", msg);
    return r;
  }

  public static R ok(Map<String, Object> map) {
    R r = new R();
    r.putAll(map);
    return r;
  }

  public static R ok() {
    return new R();
  }

  public R put(String key, Object value) {
    super.put(key, value);
    return this;
  }

  public Integer getCode() {
    return (Integer) this.get("code");
  }

  public String getMsg() {
    return (String) this.get("msg");
  }
}
